package com.epam.lissina.entity;

import com.epam.lissina.enums.TypeOfTransport;

public class MedicalTour extends Tour {

	private String kindOfTreatment;
	private int additionalCost;

	public MedicalTour(TypeOfTransport transport, double cost, int days,
			int additionalCost, int kindOfTreatment) {
		super(transport, cost, days);
		setKindOfTreatment(kindOfTreatment);
		this.additionalCost = additionalCost;
	}

	public int getAdditionalCost() {
		return additionalCost;
	}

	public void setAdditionalCost(int addCost) {
		this.additionalCost = addCost;
	}

	public String getKindOfTreatment() {
		return kindOfTreatment;
	}

	public void setKindOfTreatment(int kindOfTreatment) {
		switch (kindOfTreatment) {
		case 1:
			this.kindOfTreatment = "fangotherapy";
			break;
		case 2:
			this.kindOfTreatment = "massage";
			break;
		case 3:
			this.kindOfTreatment = "wraps";
			break;
		case 4:
			this.kindOfTreatment = "injections";
			break;
		default:
			this.kindOfTreatment = "Other";
		}
	}

	@Override
	public String toString() {
		return "Treatment [Kind of treatment " + kindOfTreatment
				+ ", additionalCost " + additionalCost + ", Transport "
				+ getTransport() + ", Cost " + getCost() + ", Days "
				+ getDays() + "]";
	}
}
